package com.baidu.leetcode.datastructures.sparsearray;

/**
 * @author shilei
 * @create 2021-10-08 14:14
 * 五子棋案例演示
 * 原始二维数组转稀疏数组案例演示
 */
public class SparseArray {
    public static void main(String[] args) {
        //创建一个原始的二维数组 11 * 11
        //0 表示没有棋子  1表示黑子 2表示蓝子
        int chessArr[][] = new int[11][11];
        chessArr[1][2] = 1;
        chessArr[2][3] = 2;
        chessArr[4][5] = 1;
        System.out.println("原始二维数组为：");
        for (int[] item : chessArr) {
            for (int data : item) {
                System.out.printf("%d\t", data);
            }
            System.out.println();
        }

        //1.获取原始二维数组有效值总数 sum
        int sum = 0;
        for (int i = 0; i < chessArr.length; i++) {
            for (int j = 0; j < chessArr.length; j++) {
                if (chessArr[i][j] != 0) {
                    sum++;
                }
            }
        }
        System.out.println("sum=" + sum);

        //2.创建稀疏数组
        int sparseArr[][] = new int[sum + 1][3];

        //3.给稀疏数组赋值
        sparseArr[0][0] = chessArr.length;
        sparseArr[0][1] = chessArr.length;
        sparseArr[0][2] = sum;

        //用于记录有效数据是第几个
        int count = 0;
        for (int i = 0; i < chessArr.length; i++) {
            for (int j = 0; j < chessArr.length; j++) {
                if (chessArr[i][j] != 0) {
                    count++;
                    //稀疏数组只有固定三列  1列为原始数组的行数 一列为原始数据的列数  一列为原始数组的值
                    sparseArr[count][0] = i;
                    sparseArr[count][1] = j;
                    sparseArr[count][2] = chessArr[i][j];
                }
            }
        }

        System.out.println("稀疏数组:");
        //打印稀疏数组验证结果
        for (int i = 0; i < sparseArr.length; i++) {
            System.out.printf("%d\t%d\t%d\t\n", sparseArr[i][0], sparseArr[i][1], sparseArr[i][2]);
        }

        System.out.println();
        //将稀疏数组还原为原始数组
        //1.先读取稀疏数组的第一行 根据第一行的数据 创建原始二维数组
        int chess2Arr[][] = new int[sparseArr[0][0]][sparseArr[0][1]];
        for (int i = 1; i < sparseArr.length; i++) {
            chess2Arr[sparseArr[i][0]][sparseArr[i][1]] = sparseArr[i][2];
        }

        //打印恢复后的二维数组
        for (int[] item : chess2Arr) {
            for (int data : item) {
                System.out.printf("%d\t", data);
            }
            System.out.println();
        }
    }
}
